package com.lingjtx.common.ip;

import cn.hutool.core.io.resource.ResourceUtil;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.io.IOException;

/**
 * IP 工具类
 *
 * @author John Liu
 */
public class IPUtils {

    private static Logger log = LoggerFactory.getLogger(IPUtils.class);

    /**
     * 初始化 SEARCHER
     */
    public final static IPUtils INSTANCE = new IPUtils();

    /**
     * IP 查询器，启动加载到内存中
     */
    private static Searcher SEARCHER;

    /**
     * 私有化构造
     */
    private IPUtils() {
        try {
            byte[] bytes = ResourceUtil.readBytes("ip2region.xdb");
            SEARCHER = Searcher.newWithBuffer(bytes);
        } catch (IOException e) {
            log.error("启动加载 IPUtils 失败", e);
        }
    }

    /**
     * 查询 IP 对应的地区编号
     *
     * @param ip IP 地址，格式为 127.0.0.1
     * @return 地区id
     */
    public static Integer getAreaId(String ip) {
        try {
            return Integer.parseInt(SEARCHER.search(ip.trim()));
        } catch (Exception e) {
            log.error("查询 IP 对应的地区编号异常：{}",e);
            return null;
        }
    }

    /**
     * 查询 IP 对应的地区编号
     *
     * @param ip IP 地址的时间戳，格式参考{@link Searcher#checkIP(String)} 的返回
     * @return 地区编号
     */
    public static Integer getAreaId(long ip) {
        try {
            return Integer.parseInt(SEARCHER.search(ip));
        } catch (IOException e) {
            log.error("查询 IP 对应的地区编号异常：{}",e);
            return null;
        }
    }

    /**
     * 查询 IP 对应的地区
     *
     * @param ip IP 地址，格式为 127.0.0.1
     * @return 地区
     */
    public static Area getArea(String ip) {
        return AreaUtils.getArea(getAreaId(ip));
    }

    /**
     * 查询 IP 对应的地区
     *
     * @param ip IP 地址的时间戳，格式参考{@link Searcher#checkIP(String)} 的返回
     * @return 地区
     */
    public static Area getArea(long ip) {
        return AreaUtils.getArea(getAreaId(ip));
    }
}
